import torch
from SepMark_main.utils.mask_img_loader import maskImgDataset